# 导入 flask 依赖
from flask import Flask, render_template, request
import numpy as np # 导入 numpy 模块
import db # 导入自定义的 db 模块
import charts # 导入自定义的 charts 模块

# 创建 flask 实例
app = Flask(__name__)

# 定义路由
@app.route('/')
def hello():
    return render_template("index.html")

@app.route("/search", methods=["POST"])
def search():
    # 在 flask 中如何接收表单数据 ?
    # request.form['code']
    code = request.form.get("code")
    start_date = request.form.get("start_date")
    end_date = request.form.get("end_date")
    data = db.query(code, start_date, end_date)
    # 将数据转换为 list 格式
    arr = np.array(data).tolist()

    chart = "abcd";
    return render_template("chart.html", data=arr, chart=chart)

@app.route("/search", methods=["Get"])
def search1():
    # 在 flask 中如何接收表单数据?
    # request.form['code']
    code = request.args.get("code")
    start_date = request.args.get("start_date")
    end_date = request.args.get("end_date")
    # 查询数据 开盘价数据
    data = db.query(code, start_date, end_date)
    # 开盘价折线图
    chart = charts.chart(data, start_date, end_date)
    # 生成 K 线图
    k_chart = charts.k(data, start_date, end_date)


    return render_template("chart.html", chart=chart, k_chart=k_chart)

@app.route("/hello")
def hello2():
    # 响应字符串
    return "Hello Flask!"

@app.route("/html")
def html():
    # 响应 html 文件
    return render_template("demo.html")

# 启动 flask 服务
if __name__ == '__main__':
    app.run()

